#!/bin/bash
#SBATCH -J test                         #
#SBATCH -p xahdexclu04                  #
#SBATCH -N 1                            #
#SBATCH --ntasks-per-node=1             #
#SBATCH -c 8                            #
#SBATCH --gres=dcu:1                    #
#SBATCH -o %j.out                       #
#SBATCH -e %j.out


module purge
module load anaconda3/5.2.0
module load compiler/devtoolset/7.3.1
module load mpi/hpcx/gcc-7.3.1
module load compiler/dtk/22.10
module list

source activate jassor

~/.conda/envs/jassor/bin/python main/start_train.py --no-mps